import pandas as pd
import os
from datetime import datetime
import xlsxwriter


def 数据写入(data, region_code):
    # with open('导出.txt', 'r', encoding='utf-8') as f:
    #     headers = json.load(f)
    # # 假设你的数据存储在一个列表中
    # data =headers['data']

    # 提取所需的字段
    extracted_data = [
        {'号码': item['getnum_no'], '窗口号': item['ck_number'],
         '姓名': item['getnum_xname'],
         '手机号码': item['getnum_tel'], '证件号码': item['getnum_idc'], '取号时间': item['getnum_time'],
         '数字优先级': item['priority_num'], '窗口名称': item['ck_name'], '部门名称': item['org_name'],
         '业务名称': item['yw_name'], '叫号人': item['call_user_xname'], '是否预约号': '',
         '叫号时间': item['call_time'],
         '结束时间': item['end_time'], '是否续办号': '', '续办时间': '', '是否特呼': '', '是否弃号': '', '数据来源': '',
         '添加时间': item['create_date']
         }
        for item in data]
    # 创建DataFrame
    df = pd.DataFrame(extracted_data)
    # 获取当前年月日
    current_date = datetime.now().strftime('%Y%m%d')+'取号记录查询'

    # 目标目录路径


    # 如果目录不存在，则创建
    if not os.path.exists(current_date):
        os.makedirs(current_date)

    # 完整文件路径
    file_path = os.path.join(current_date, f'{region_code}.xlsx')

    # 写入 Excel 文件
    with pd.ExcelWriter(file_path, engine='xlsxwriter') as writer:
        df.to_excel(writer, index=False, sheet_name='Sheet1')

    print(f"数据已成功写入 {region_code}.xlsx")

def 数据写入2(data, region_code):
    # with open('导出.txt', 'r', encoding='utf-8') as f:
    #     headers = json.load(f)
    # # 假设你的数据存储在一个列表中
    # data =headers['data']

    # 提取所需的字段
    extracted_data = [
        {'号码': item['getnum_no'], '窗口号': item['ck_number'],
         '姓名': item['getnum_xname'],
         '手机号码': item['getnum_tel'], '证件号码': item['getnum_idc'], '取号时间': item['getnum_time'],
         '数字优先级': item['priority_num'], '窗口名称': item['ck_name'], '部门名称': item['org_name'],
         '业务名称': item['yw_name'], '叫号人': item['call_user_xname'], '是否预约号': '',
         '叫号时间': item['call_time'],
         '结束时间': item['end_time'], '是否续办号': '', '续办时间': '', '是否特呼': '', '是否弃号': '', '数据来源': '',
         '添加时间': item['create_date']
         }
        for item in data]
    # 创建DataFrame
    df = pd.DataFrame(extracted_data)
    # 获取当前年月日
    current_date = datetime.now().strftime('%Y%m%d')+'取号记录查询'

    # 目标目录路径


    # 如果目录不存在，则创建
    if not os.path.exists(current_date):
        os.makedirs(current_date)

    # 完整文件路径
    file_path = os.path.join(current_date, f'{region_code}.xlsx')

    # 写入 Excel 文件
    with pd.ExcelWriter(file_path, engine='xlsxwriter') as writer:
        df.to_excel(writer, index=False, sheet_name='Sheet1')

    print(f"数据已成功写入 {region_code}.xlsx")